www.gusucode.com > VC++ 同学录信息管理系统源代码-源码程序 > VC++ 同学录信息管理系统源代码-源码程序/code/UserDlg.cpp

    //Download by http://www.NewXing.com
// UserDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Mates.h"
#include "UserDlg.h"

#include "AboutDialog.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CUserDlg dialog
static UINT indicators[] =
{
	IDS_STATUS_DATA,		
	ID_SEPARATOR,           
	ID_INDICATOR_CAPS,
	ID_INDICATOR_NUM,
	ID_INDICATOR_SCRL,
	IDS_STATUS_SPACE,				
};

CUserDlg::CUserDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CUserDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CUserDlg)
	m_strUserName = _T("");
	m_strPower = _T("");
	m_strPassWord = _T("");
	m_strTel = _T("");
	//}}AFX_DATA_INIT
	m_TreeBootImage.Create (IDB_BITMAP_TREE,20,1,ILC_COLOR32);  
}


void CUserDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CUserDlg)
	DDX_Control(pDX, IDC_EDIT_NAME, m_ctrlName);
	DDX_Control(pDX, IDC_LIST, m_ctrlList);
	DDX_Control(pDX, IDC_TREE, m_ctrlTree);
	DDX_Text(pDX, IDC_EDIT_NAME, m_strUserName);
	DDX_Text(pDX, IDC_EDIT_POWER, m_strPower);
	DDX_Text(pDX, IDC_EDIT_PW, m_strPassWord);
	DDX_Text(pDX, IDC_EDIT_TEL, m_strTel);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CUserDlg, CDialog)
	//{{AFX_MSG_MAP(CUserDlg)
	ON_COMMAND(ID_REFRESH, OnRefresh)
	ON_NOTIFY(TVN_SELCHANGED, IDC_TREE, OnSelchangedTree)
	ON_NOTIFY(LVN_COLUMNCLICK, IDC_LIST, OnColumnclickList)
	ON_NOTIFY(NM_CLICK, IDC_LIST, OnClickList)
	ON_COMMAND(ID_CLEAR, OnClear)
	ON_COMMAND(ID_HELP, OnHelp)
	ON_BN_CLICKED(ID_SEARCH, OnSearch)
	ON_BN_CLICKED(ID_DELETE, OnDelete)
	ON_COMMAND(ID_QUIT, OnQuit)
	ON_COMMAND(ID_VERSION, OnVersion)
	ON_BN_CLICKED(ID_ADD, OnAdd)
	ON_BN_CLICKED(ID_EDIT, OnEdit)
	ON_WM_CONTEXTMENU()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CUserDlg message handlers

BOOL CUserDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// TODO: Add extra initialization here
	
	CRect	rect;
	GetClientRect(rect);	

	m_wndStatusBar.Create(this);
	m_wndStatusBar.SetIndicators(indicators,sizeof(indicators)/sizeof(UINT));
	m_wndStatusBar.SetPaneStyle(2,SBPS_STRETCH);
	m_wndStatusBar.MoveWindow(0,rect.bottom-20,rect.right,20);

	
	m_ctrlTree.SetImageList(&m_TreeBootImage,TVSIL_NORMAL);	
	m_ctrlList.ModifyStyle(0,LVS_REPORT|LVS_SHOWSELALWAYS|LVS_SINGLESEL);
	m_ctrlList.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES|LVS_EX_HEADERDRAGDROP);	
	m_pRS.ADOOpen();    
	CUserDlg::OnRefresh(); 

	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CUserDlg::OnRefresh() 
{
	// TODO: Add your command handler code here
	m_strSQL="SELECT * FROM UserTab";
	m_pRS.ADOExcute(m_strSQL);
	AddTreeNodes(); 
	AddListItems();	
}


void CUserDlg::AddTreeNodes(){

	m_ctrlTree.DeleteAllItems();	       


	hRootItem=m_ctrlTree.InsertItem("用户管理信息",0,2,TVI_ROOT,TVI_LAST);
	m_pRS.MoveFirst();	
	
	while(!m_pRS.ADOEOF())
	{
	
		HTREEITEM hNative=AddDistinctNode(hRootItem,m_pRS.GetFieldString("UserName"));
		m_pRS.MoveNext();
	}
}


HTREEITEM	CUserDlg::AddDistinctNode(HTREEITEM hParent,CString strValue)
{
	HTREEITEM hItem=m_ctrlTree.GetChildItem(hParent);
	while(hItem)
	{
		CString strItemText = m_ctrlTree.GetItemText(hItem);
		if(strItemText == strValue)
	return hItem;
		hItem = m_ctrlTree.GetNextSiblingItem(hItem);
	}

	return m_ctrlTree.InsertItem(strValue,1,2,hParent,TVI_LAST);
}
void CUserDlg::AddListItems()
{
	int nCount=0;
	CHeaderCtrl *pHeaderCtrl=m_ctrlList.GetHeaderCtrl();
	if(pHeaderCtrl!=NULL)
		nCount=pHeaderCtrl->GetItemCount();
	for(int i=0;i<nCount;i++)
		m_ctrlList.DeleteColumn(0);
	m_ctrlList.DeleteAllItems();

	for(i=0;i<m_pRS.nFieldCols;i++)
	{
		m_ctrlList.InsertColumn(i,m_pRS.GetFieldName(i),LVCFMT_LEFT,90);
	}
	int nItem=0;
	while(!m_pRS.ADOEOF()){
		m_ctrlList.InsertItem(nItem,m_pRS.GetFieldString(0));
		for(int i=1;i<m_pRS.nFieldCols;i++)
			m_ctrlList.SetItemText(nItem,i,m_pRS.GetFieldString(i));	
		m_pRS.MoveNext();
	}
}

void CUserDlg::OnSelchangedTree(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;	
	*pResult = 0;
	HTREEITEM hSelItem=m_ctrlTree.GetSelectedItem();	

	HTREEITEM hParentItem=m_ctrlTree.GetParentItem(hSelItem); 
	CString strSql,str,strCondition;
	strSql="SELECT * FROM UserTab";
	if(hParentItem!=NULL)
	{
		HTREEITEM hOldParent=m_ctrlTree.GetParentItem(hParentItem);
		if(hOldParent!=NULL)
		{
			strCondition.Format(" Where UserName='%s' AND UserName='%s'",m_ctrlTree.GetItemText(hParentItem),m_ctrlTree.GetItemText(hSelItem));  //此为一行
		}
		else
		{
	
			strCondition.Format(" Where UserName='%s'",m_ctrlTree.GetItemText(hSelItem));
		}	
	}
	else
	{
		strCondition="";  
	}
	strSql+=strCondition;  
	m_strSQL=strSql;	    
	m_pRS.ADOExcute(strSql);  
	AddListItems();           
}

void CUserDlg::OnColumnclickList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
	// TODO: Add your control notification handler code here
	
	*pResult = 0;
	CPoint   pt;   
	::GetCursorPos(&pt);
	m_ctrlList.ScreenToClient(&pt);   
	LVHITTESTINFO Info;   
	Info.pt = pt;   
	Info.flags = LVHT_ABOVE;
	m_ctrlList.SubItemHitTest(&Info);	

	LVCOLUMN   col;	
	col.mask=LVCF_TEXT;   
	col.cchTextMax =20;
	char   buff[20];   
	col.pszText=buff;
	m_ctrlList.GetColumn(Info.iSubItem,&col);	

	CString strSQL=m_strSQL+" ORDER BY "+col.pszText;
	m_pRS.ADOExcute(strSQL);   
	AddListItems();             
}

void CUserDlg::OnClickList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	// TODO: Add your control notification handler code here
	m_ctrlName.EnableWindow(false);
	int i=m_ctrlList.GetSelectionMark();     
	m_strUserName=m_ctrlList.GetItemText(i,0);
	m_strPassWord=m_ctrlList.GetItemText(i,1);
	m_strPower=m_ctrlList.GetItemText(i,2);
	m_strTel=m_ctrlList.GetItemText(i,3);
	UpdateData(false);	 
	*pResult = 0;
}


void CUserDlg::OnClear() 
{
	// TODO: Add your command handler code here
	m_ctrlName.EnableWindow(true);
	m_strUserName="";
	m_strPassWord="";
	m_strPower="";
	m_strTel="";
	UpdateData(false);
}

void CUserDlg::OnHelp() 
{
	// TODO: Add your command handler code here
	MessageBox("HELP!");
}

void CUserDlg::OnSearch() 
{
	// TODO: Add your command handler code here
	UpdateData(true);

	if (!m_strUserName.IsEmpty()) {
	m_pRS.ADOExcute("SELECT * FROM UserTab WHERE UserName='"+m_strUserName+"'");
	AddListItems();
	UpdateData(false);
	}
	else
		AfxMessageBox("请输入名字");
}


void CUserDlg::OnDelete() 
{
	// TODO: Add your control notification handler code here
	if(m_ctrlList.GetSelectionMark()<0)
	{
		AfxMessageBox("请先选中要删除的项!");
		return;
	}
	if(AfxMessageBox("您确认要删除吗?",MB_OKCANCEL)==IDOK)
	{
		UpdateData(true);
		CString sql;
		sql.Format("DELETE * FROM UserTab WHERE UserName='%s'",m_strUserName);
		if(m_pRS.ADOExcuteNoQuery(sql)==1)
		{	
			AfxMessageBox("删除记录成功!");
		
			m_ctrlList.DeleteItem(m_ctrlList.GetSelectionMark());
		}
		else
		{
			AfxMessageBox("删除记录失败!");
		}
	}
}

void CUserDlg::OnQuit() 
{
	// TODO: Add your command handler code here
	CDialog::OnOK();
}

void CUserDlg::OnVersion() 
{
	// TODO: Add your command handler code here
	CAboutDialog dlg;
	dlg.DoModal();
}

void CUserDlg::OnAdd() 
{
	// TODO: Add your control notification handler code here
	UpdateData(true);
	CString sql;
//	CString str;
	sql.Format("INSERT INTO UserTab (UserName,[PassWord],Userpower,UserTel) Values('%s','%s','%s','%s')",m_strUserName,m_strPassWord,m_strPower,m_strTel); 
	if (!m_strUserName.IsEmpty())  {
//		str.Format("%d",m_pRS.ADOExcuteNoQuery(sql));
//		AfxMessageBox(str);
	if(m_pRS.ADOExcuteNoQuery(sql)==1)
	{	
		AfxMessageBox("授权用户成功!");
		int n=m_ctrlList.GetItemCount();
		n=m_ctrlList.InsertItem(n,m_strUserName);			
		m_ctrlList.SetItemText(n,1,m_strPassWord);
		m_ctrlList.SetItemText(n,2,m_strPower);
		m_ctrlList.SetItemText(n,3,m_strTel);
	}
	else
	{
		AfxMessageBox("授权失败,可能是重复的用户名!");
	}
	}
	else
	{
		AfxMessageBox("授权用户失败,请输入用户资料!");
	}
}

void CUserDlg::OnEdit() 
{
	// TODO: Add your control notification handler code here
	UpdateData(true);
	CString sql;
	sql.Format("UPDATE UserTab SET [PassWord]='%s',Userpower='%s',UserTel='%s' WHERE UserName='%s'",m_strPassWord,m_strPower,m_strTel,m_strUserName);
	if (!m_strUserName.IsEmpty())  {
	if(m_pRS.ADOExcuteNoQuery(sql)==1)
	{	
		AfxMessageBox("编辑记录成功!");
		m_pRS.ADOExcute(m_strSQL);  
		AddListItems();
	}
	else
	{
		AfxMessageBox("编辑失败!");
	}
	}
	else
	{
		AfxMessageBox("编辑记录失败!");
	}
}

void CUserDlg::OnContextMenu(CWnd* pWnd, CPoint point) 
{
	// TODO: Add your message handler code here
	CRect rect;
	GetClientRect(&rect);
	CMenu menu;
	menu.LoadMenu(IDR_MENU_USER);
	CMenu* pConTextMenu=menu.GetSubMenu(0);
	GetCursorPos(&point);
	pConTextMenu->TrackPopupMenu(TPM_LEFTALIGN|TPM_LEFTBUTTON,point.x,point.y,this,&rect);
	menu.DestroyMenu();
}